package com.atguigu.tingshu.user.service.impl;

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.util.IdUtil;
import com.alibaba.fastjson.JSON;
import com.atguigu.tingshu.common.constant.KafkaConstant;
import com.atguigu.tingshu.common.constant.RedisConstant;
import com.atguigu.tingshu.common.constant.SystemConstant;
import com.atguigu.tingshu.common.service.KafkaService;
import com.atguigu.tingshu.common.util.AuthContextHolder;
import com.atguigu.tingshu.common.util.MongoUtil;
import com.atguigu.tingshu.model.user.UserListenProcess;
import com.atguigu.tingshu.user.service.UserListenProcessService;
import com.atguigu.tingshu.vo.album.TrackStatMqVo;
import com.atguigu.tingshu.vo.user.UserListenProcessVo;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.Sort;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.data.mongodb.core.query.Criteria;
import org.springframework.data.mongodb.core.query.Query;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Service;

import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeUnit;

@Service
@SuppressWarnings({"all"})
public class UserListenProcessServiceImpl implements UserListenProcessService {

    @Autowired
    private MongoTemplate mongoTemplate;

    @Autowired
    private RedisTemplate redisTemplate;

    @Autowired
    private KafkaService kafkaService;

    /**
     * 获取当前用户声音播放进度
     *
     * @param userId  用户ID
     * @param trackId 声音ID
     * @return
     */
    @Override
    public BigDecimal getTrackBreakSecond(Long userId, Long trackId) {
        //1.构建查询对象 查询条件：用户ID、声音ID 等值
        Query query = new Query();
        query.addCriteria(Criteria.where("userId").is(userId).and("trackId").is(trackId));
        //2.执行查询 手动指定用户播放进度集合名称  userListenProcess_用户ID

        //mongoTemplate.findOne(query, UserListenProcess.class, MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS.getCollectionPrefix()+"_"+userId)；
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
        if (userListenProcess != null) {
            return userListenProcess.getBreakSecond();
        }
        return new BigDecimal("0.00");
    }


    /**
     * 更新声音播放进度及播放统计信息
     *
     * @param userId              用户ID
     * @param userListenProcessVo 声音信息
     * @return
     */
    @Override
    public void updateListenProcess(Long userId, UserListenProcessVo userListenProcessVo) {
        //1.根据用户ID+声音ID查询播放进度
        Query query = new Query(Criteria.where("userId").is(userId).and("trackId").is(userListenProcessVo.getTrackId()));
        UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));

        //1.1 如果存在则修改进度、更新时间
        BigDecimal breakSecond = BigDecimal.valueOf(userListenProcessVo.getBreakSecond());
        breakSecond.setScale(2, RoundingMode.HALF_UP);
        if (userListenProcess != null) {
            userListenProcess.setBreakSecond(breakSecond);
            userListenProcess.setUpdateTime(new Date());
        } else {
            //1.2 如果不存在则新增播放进度
            userListenProcess = new UserListenProcess();
            userListenProcess.setUserId(userId);
            userListenProcess.setAlbumId(userListenProcessVo.getAlbumId());
            userListenProcess.setTrackId(userListenProcessVo.getTrackId());
            userListenProcess.setBreakSecond(breakSecond);
            userListenProcess.setIsShow(1);
            userListenProcess.setCreateTime(new Date());
            userListenProcess.setUpdateTime(new Date());
        }
        mongoTemplate.save(userListenProcess, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));

        //2.异步基于MQ更新声音统计信息

        //2.1 确保24小时内/当前0点前播放进度只统计一次 基于Redis实现
        //2.1.1 构建避免用户多次对声音进行统计Key 形式：前缀+用户ID:声音ID
        String key = RedisConstant.USER_TRACK_REPEAT_STAT_PREFIX + userId + ":" + userListenProcessVo.getTrackId();
        //2.1.2 调用setnx命令对应api
        Boolean flag = redisTemplate.opsForValue().setIfAbsent(key, userListenProcessVo.getTrackId().toString(), 24, TimeUnit.HOURS);
        //2.1.3 当存入Key成功-说明24小时内第一次统计
        if (flag) {
            //2.2 发送MQ消息到Kafka话题
            //2.2.1 创建用于更新声音统计信息VO对象
            TrackStatMqVo mqVo = new TrackStatMqVo();
            //产生消息唯一标识，用于消费者端幂等性判断处理
            mqVo.setBusinessNo(IdUtil.fastSimpleUUID());
            mqVo.setAlbumId(userListenProcessVo.getAlbumId());
            mqVo.setTrackId(userListenProcessVo.getTrackId());
            mqVo.setStatType(SystemConstant.TRACK_STAT_PLAY);
            mqVo.setCount(1);
            //2.2.2 发送消息
            kafkaService.sendMessage(KafkaConstant.QUEUE_TRACK_STAT_UPDATE, JSON.toJSONString(mqVo));
        }
    }

    /**
     * 获取当前登录用户上次播放声音记录，用于快速播放
     *
     * @return
     */
    @Override
    public Map<String, Long> getLatelyTrack() {
        //1.获取用户ID 用户ID不为空再查询
        Long userId = AuthContextHolder.getUserId();
        if (userId != null) {
            //2.构建查询条件：1.用户ID  按照：更新时间进行倒序 获取第一条记录
            Query query = new Query(Criteria.where("userId").is(userId));
            query.with(Sort.by(Sort.Direction.DESC, "updateTime"));
            query.limit(1);
            UserListenProcess userListenProcess = mongoTemplate.findOne(query, UserListenProcess.class, MongoUtil.getCollectionName(MongoUtil.MongoCollectionEnum.USER_LISTEN_PROCESS, userId));
            if (userListenProcess != null) {
                //3.封装最近播放专辑ID 及 声音ID
                Map<String, Long> mapResult = new HashMap<>();
                mapResult.put("albumId", userListenProcess.getAlbumId());
                mapResult.put("trackId", userListenProcess.getTrackId());
                return mapResult;
            }
        }
        return null;
    }

}
